package com.plm.maque;

import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import java.util.ArrayList;
import java.util.List;

/**
 * @description 代码生成器
 * @author Limi
 * @date 2018-12-25
 */
public class Generator {
    /**数据库url*/
    private static String dbUrl = "jdbc:mysql://localhost:3306/maque?serverTimezone=GMT";
    /**数据库驱动*/
    private static String dbDriver = "com.mysql.cj.jdbc.Driver";
    /**数据库登录用户名*/
    private static String dbUsername = "root";
    /**数据库登录密码*/
    private static String dbPassword = "root@2018";
    /**类注释作者*/
    private static String author = "Limi Pan";
    /**包父目录*/
    private static String packageParent = "com.plm.maque";
    /**数据库表前缀*/
    private static String tablePrefix = "sys_";
    /**根据哪张表生成代码*/
    private static String tableName = "sys_user";

    public static void main(String[] args) {
        // 代码生成器
        AutoGenerator mpg = new AutoGenerator();

        // 全局配置
        GlobalConfig gc = new GlobalConfig();
        String projectPath = System.getProperty("user.dir"); //获取项目根路径
        gc.setOutputDir(projectPath + "/src/main/java"); //生成路径
        gc.setAuthor(author); //设置作者
        gc.setFileOverride(true); //设置生成文件是否覆盖
        gc.setOpen(false); //设置是否打开
        gc.setServiceName("%sService"); //生成service接口的名字的首字母是否为IUserService
        gc.setBaseResultMap(true); //设置SQL映射文件
        gc.setBaseColumnList(true); //设置SQL映射文件中基本列
        mpg.setGlobalConfig(gc);

        // 数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl(dbUrl);
        dsc.setDriverName(dbDriver);
        dsc.setUsername(dbUsername);
        dsc.setPassword(dbPassword);
        mpg.setDataSource(dsc);

        // 包配置
        PackageConfig pc = new PackageConfig();
        pc.setParent(packageParent)
                .setMapper("dao");
        //.setModuleName(scanner("模块名"))
        //.setController("controller")
        //.setEntity("entity")
        //.setService("service")
        //.setXml("mapper");
        mpg.setPackageInfo(pc);

        // 自定义配置
        InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                // to do nothing
            }
        };
        List<FileOutConfig> focList = new ArrayList<>();
        focList.add(new FileOutConfig("/templates/mapper.xml.ftl") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定义输入文件名称
                return projectPath + "/src/main/resources/mapper/" /*+ pc.getModuleName()+ "/" */
                        + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
            }
        });
        cfg.setFileOutConfigList(focList);
        mpg.setCfg(cfg);
        mpg.setTemplate(new TemplateConfig().setXml(null));

        // 策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setNaming(NamingStrategy.underline_to_camel); //数据库 表 映射到实体的命名策略:下划线转化为驼峰命式
        strategy.setColumnNaming(NamingStrategy.underline_to_camel); //数据库 表字段 映射到实体的命名策略：下划线转化为驼峰命式
        //strategy.setSuperEntityClass("com.plm.mybms.entity"); //自定义继承的Entity类全称，带包名
        strategy.setEntityLombokModel(true); //【实体】是否为lombok模型（默认 false）
        strategy.setRestControllerStyle(true); //设置Rest风格的Controller
        //strategy.setSuperControllerClass("com.plm.mybms.controller"); //自定义继承的Controller类全称，带包名
        strategy.setInclude(tableName); //需要包含的表名，允许正则表达式（与exclude二选一配置）
        //strategy.setSuperEntityColumns("id"); //自定义基础的Entity类，公共字段
        strategy.setControllerMappingHyphenStyle(true); // 驼峰转连字符
        strategy.setTablePrefix(tablePrefix); //设置表前缀
        mpg.setStrategy(strategy);

        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
        mpg.execute();
    }
}
